Matrices and Vectors with Python | How to reshape a matrix